package cp4.resp;

import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

public class ShortUrlServlet extends HttpServlet {
    // 短链接的映射
    private Map<String, String> urlMap = new HashMap<>();

    @Override
    public void init(ServletConfig config) throws ServletException {
        super.init(config);
        // 初始化链接映射
        urlMap.put("baidu", "https://www.baidu.com");
        urlMap.put("qq", "https://www.qq.com");
        urlMap.put("weibo", "https://www.weibo.com");
        urlMap.put("b", "https://www.bilibili.com");
    }

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        // url?id=baidu 或者 url?id=qq
        String id = req.getParameter("id");// 获取请求的参数
        if (id == null || !urlMap.containsKey(id)) {
            // 如果没有参数 或者没有对应的设置链接 直接报错
            resp.sendError(404);
        } else {
            // 正常情况
            String url = urlMap.get(id); // 根据key获取到映射的完整链接
            // 跳转
            // 实质是:
            // resp.setStatus(302);
            // resp.setHeader("Location",url);
            resp.sendRedirect(url);
        }

    }
}
